package com.db.util;

import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;

import javax.servlet.http.HttpServletRequest;
import java.io.File;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.UUID;

public class FileUploadUitl {
    public static Map<String,String> upload(HttpServletRequest request){
        try {
            //创建文件仓库，用于存储用户上传的文件
            DiskFileItemFactory fileFactory = new DiskFileItemFactory();

            //获取工件中的upload_repository目录路径
            String realPath = request.getServletContext().getRealPath("upload_repository");
            //创建表示仓库目录的文件对象
            File repository = new File(realPath);
            //指定仓库目录的位置
            fileFactory.setRepository(repository);

            //准备接收数据,分配仓库
            ServletFileUpload fileUpload = new ServletFileUpload(fileFactory);
            //解析请求中所存储的数据（文件的内容与表单的参数）
            //返回值是表单中的所有表单项
            List<FileItem> items = fileUpload.parseRequest(request);
            //创建Map集合，用于存储普通表单中的数据
            Map<String, String> map = new HashMap<>();
            String fileName = null;
            for (FileItem item : items) {
                //判断是否为普通表单
                if (item.isFormField()) {
                    //普通表单
                    String name = item.getFieldName();
                    System.out.println(name);
                    String value = item.getString("UTF-8");
                    //将数据封装到Map中 以表单name为key，value为value
                    map.put(name, value);

                } else {
                    //文件表单
                    //拼接文件输出路径
                    String rawName = item.getName();
                    if(!"".equals(rawName)){
                        fileName = UUID.randomUUID().toString() + rawName.substring(rawName.lastIndexOf('.'));
                        String filePath = realPath + "\\" + fileName;
                        
                        //构建文件对象
                        File uploadFile = new File(filePath);

                        //将表单中的文件数据写到uploadFile中
                        item.write(uploadFile);

                    }


                }
                //String name = item.getName();//如果此项表示文件，则name为文件名否则为null
                //String name =item.getFieldName();//如果此项是普通表单项。则name为表单参数的key
                //String value = item.getString();//获取表单项中的value

            }
            map.put("fileName",fileName);
            return map;
        }catch (Exception e){
            e.printStackTrace();
        }
        return null;
    }
}
